Fixed problem 2