Fix removeHighlights: custom CSS classes