Add sumOfSquares method