Added a toggle for loading data to memory in shuffle2
Merged