fixed coloring at the end