Added viewmodel