fix problem 3 answer