Fix problem 3