Update to week 6 assignment