Add partial problem 3 solution