Added prob_3_test.py