Minor improvements to code editor