Update: Added colormap function