Trying to assign colors in the DataFrame