Updates for week 2 assignment