Create pset1_problem2_counting_bobs.py