Python Week 7 Solution