Finish square_of_sum method and get test to pass.