Adding challenge 3 code