Python Week 6 solution