Add the migration class