Use your max function from task 1.
Hint: `from task_1_max import max`