Set 2, challenge 10 solution