Problem5.py V4-Final