modify pa1.py