add find_square method