assignment 7 - fix one typo in problem 2