The only reason not to do that would be for documentation reasons. Doing Pregel in both directions is theoretically breaking the Pregel abstraction (I think, correct me if I'm wrong). That being said, I think just having it as an argument is easier for the user. @jegonzal @rxin opinions?