Update Assignment1Solution.py