input: Write a python function to return the sum of two numbers instead of printing.
  assert sum_two_numbers(3, 4) == 7
  assert sum_two_numbers(10, 5) == 15
  assert sum_two_numbers(-1, 1) == 0
output: def sum_two_numbers(a, b):
      return a + b