Update challenge 4 solution