Added plot of how the gradient of the cost function varies with layer