input: Write a python function to calculate the surface area of a cube.
  assert cube_area(3) == 54
  assert cube_area(1) == 6
  assert cube_area(5) == 150
output: def cube_area(side_length):
      return 6 * side_length ** 2