added comments in lab2_hw.py