Update p1/main.py