fixed typo and addressed comment in assignment 3