Added a set function in order for the coloring to be completed