Update ps1b.py

completed problem 1b 10/3/13