[Feature Request] Improve the code editor