Update ViewModel.cs