add test models
update migrate() argument