Homework 4 solution