fixed challenge 3-5 . forgot to modify.