updates to migration and missing imports