Fixed the coloring