Python Week 12 solution