add more correct version of pipeline stage coloring