Added ViewModel classes