added support for models, partial for ops