Adding homework 4