update highlight logic