Update for Homework 4 -- Part 2