Update Prob1.1.py