Added final code for week 2 turn-in