Initial work for homework 4