Updated to new version of Week 5 assignment.