Update tests, clean up perfect_square test