Update assignment10.py