修改下viewmodel的实现