Add highlighting feature