Tree view highlight improvement