Refactor migration features with tag