Updated day 2 homework