Add documentation to gradients demo