AC problem 4 solution