Add ability to highlight current line