Update sum_squares.py