Working Version - ps2.py final copy