Update Assignment3_Problem2.py