Bug fix and add VisualTreeHelper2.