Added coloring based on data. Currently two separate prep functions. Will require change later.