adding changes for homework two