My solution for problem 3.