Add highlighting to live preview demo