input: Write a python function to compute the cost function defined as the sum of squares of two numbers.
  assert cost_function(3, 4) == 25
  assert cost_function(1, 2) == 5
  assert cost_function(0, 0) == 0
output: def cost_function(a, b):
      return a**2 + b**2