Added viewmodel to views