importing missing colormap from pyplot