changed the main method to only test square(2) once