adding changes for homework four