Added code to solve challenge 4