Create squares.py