Add in distance based coloring