[imp] Improve highlighing current line