Hey @jegonzal - we've generally been trying to put runnable programs like this in the examples package. Would it be possible to do that here? The main benefits are that (i) users understand this is not a stable API and (ii) users can launch it with `run-example` or `spark-submit` which allows them to easily do things like launch on YARN or ask for a specific number of executors when running the program.