Update assignment.py