Update assignment1.py

added few comment lines