Small update to color scheme