Update for week 5 assignment