Refactor and implement the ShuffleManager
Modify code to use refactored shuffle framework