add option for custom highlightClass