Adding working samples of AllGather and Broadcast