Add utilities for color maps
Fix documentation