Change ImagePlot to GradientPlot and fix