Add support for color coding.