Adde the color assignment

Adde the color assignment