added support for model migration