Add support for partial aggregation on Join node