Example code for Week 5