Create whole_square.py