Added ViewModel to Controller