Wrote a version of set_colormap