Add highlight class and $timeout