updated challenge 5 with more comments