Adding week 3 lesson 1 code