Changed Model to ViewModel to conform the MVVM Pattern