Update for Week 06 assignment