Trying to plot the gradient