Modifed ViewModel