Update ps1.py

Finished 2.1