Fix in problem 3