Had to add a definition for square for moved test to work