Update Assignment4_Problem3.py