Update assignment for week 2