Week 2 python homework