Update P2.py