Added in a coloring function