ViewModel class added