add test for squares
test cases now pass