add code for week one