modified prob_2.py