Add commit rollback to migration