Adding RiemannSink to Spark