Program For Square In Python 3