Add the code for Week 2