finally doing homework 4!