Add custom execute and rewind in migration