Week 6 Python Assignment