Changes for homework 2