Update for last bit of Homework 3