input: Write a python function to find the maximum of two numbers.
  assert find_max(3, 5) == 5
  assert find_max(10, 7) == 10
  assert find_max(0, 0) == 0
output: def find_max(a, b):
      if a > b:
          return a
      else:
          return b