Adding allgather and broadcast