Update ps1b.py

evan's code on the bottom