Made changes to file for homework assignment