UI Improvements to the color code