Update assignment1.py