Assignment 2 - fix typo in problem 1