Add week 3 lesson code