add comments to hw1.py