Feature/update model spec