Added assignment 5.1

Reading the API docs carefully really helped. The max function made life easier.