Fixed problem 3