Solution for challenge 4