Improve CodeMirror styling and syntax highlighting