assignment 6 - fix problem 1 typo