@shacker do we still need to add the `color_class` here?
If not, can you remove it?