Max Function With 1 Param - Python