Update the add color version

Add the color version!!