Fixed the colors of the visualization. Maybe the colors can be sorted better