Update colormap.py