square root program in python