cleaning up square example