changed highlight to properly reset color