Week 7 Python Assignment