Solution to challenge 1