[FEATURE] Adding more methods to AbstractMigration + some testing.