now trying in-memory-shuffle