Add matching colour to class diagram