Update assignment8.py