Added prob_5.py
Added prob_5_test.py