added squares.py