How To Write A Max Function In Python