Colormap and save_fig