added prob_3.py