Update p1.py