Solution for problem 8