Modified Tets for prob_5.py