Add challenge 4 solution