UI tweaks to editor