Updates for Week 05 assignment