Update ps1b.py

calculate function works, can't figure out how to get the check function to work