Python Assignment --Week 4