Program For Square In Python