finished square example and unit test