Add the color class