Added stuff from last homework