Remove unused modules
Add public addNewMigration method
Add simple test setup