Add capability to highlight lines in code