Improvements to live editor code handling