First working code for Week 7