Finishing run_migration func.
Making it so we can mark migrations.