Simplify the proof of the chain rule