Update HW3 part2-squre_sum