Update PA1new.py