missed some code for challenge 1