code for week 1