Updates to week 1 assignment