how to square a number in python