input: Write a function to calculate the sum of cubes of the first n natural numbers.
  assert sum_cubes(3) == 36
  assert sum_cubes(4) == 100
  assert sum_cubes(5) == 225
output: def sum_cubes(n):
      return sum(i**3 for i in range(1, n + 1))