python program to calculate square of a number