adding viewmodel