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