Improve documentation for map_agg