All tests for square function are passing