Homework 4 changes