Code for Week 1