added prob_2_test.py