update some comments in assign2.py