Adding allreduce