assignment 6 - fix typo in problem 1