Update pbj.py

Changed the first part to add input from the user. Still trying to figure out how to make the 2nd goal work.