Solution for challenge 7