Python Week 9 solution