input: Write a python function to compute the sum of squares of first n natural numbers.
  assert sum_of_squares(3) == 14
  assert sum_of_squares(5) == 55
  assert sum_of_squares(10) == 385
output: def sum_of_squares(n):
      return sum(i * i for i in range(1, n + 1))