Add tag highlighting