Update ViewModel.java