remove .reverse() on the colormap for diffusion eqn