Updates for Homework 4