Updated file for homework