square.py