Move migration to core gem