Определение наибольшего элемента в массиве

Реализовать функцию, которая принимает массив чисел и возвращает наибольшее значение из него. При желании можно переименовать параметры, но сам код функции должен остаться без изменений.

def find_max(arr):
    max_val = arr[0]
    for num in arr:
        if num > max_val:
            max_val = num
    return max_val