Filling in assignment

I filled in the code that was needed for the assignment.