Add highlighting function