refactor(models): connect with viewmodels