fix problem 3