I think map works. Haven't had a chance to test with multiple workers yet though (I think I need to fix input split partitioning before I can do that).
Fix merge conflict.