Update TreeViewModel.cs