Cleanup.
Move highlighting functionality out from CodeEditor.