Update to week 8 assignment