assignment 7 - fix typo in problem 4