Add in-class code for Week 2