Написать функцию, которая возвращает максимальное из двух целых чисел, полученных в качестве аргумента

#Python 3.4

def max(a, b):
    if a > b:
        return a
    else:
        return b

# Тестовый прогон

print(max(3, 5))
5

print(max(5, 3))
5

print(max(int(input()), int(input())))
А тут будет выведено максимальное из того, что введете