Adding color assignment