Add highlight class