add colors
minor updates