Update Assignment 5.2.py