Added code for Week 3