Week 1 Homework problem