Put solution to challenge 2