Text classification with Spark 2.0