Add necessary comments for ShuffleBlockManager