Update code to use the native CodeEditor