Updated ViewModel