Improved highlighting for live editing