Python Week 4 Solution