Design a program in Python that finds the maximum of two numbers.
number1 = 4
number2 = 6
def find_max(num1, num2):
  if num1 > num2:
    return num1
  else:
    return num2

max_num = find_max(number1, number2)
print(max_num)  # Outputs 6