Add solution to challenge 2