HW5.py Q2 Finis