Switch to thinner Scala API, DataSet now wraps Java DataSet