Using Max Function In Python