Updates for Week 05 assignment 2