Added potential solution for Problem 6