typo fix in assignment 2