Python Week 10 solution