Add solution to problem 4, part a