Solution involving chain rule