Updates for Week 05 assignment 3