Add the code for Week 3