Move highlight template to JS