UI improvements for editor