Add optional `arg` parameter to map & reduce phases