fix problem 7 answer