change color schemes to colormaps from matplotlib.cm for 2D