Update Square.py