Improvements to UI.
Fix new coloring callback.