Updates for Homework 4. Work in Progress