Add highlight check