Add solution for set 1 challenge 3