@@ -11,6 +11,7 @@ import org.apache.spark.rdd.RDD
import org.apache.spark.sql.types.{DataTypes, StructField}
import org.apache.spark.sql.{Dataset, Row, RowFactory, SparkSession}
+
import scala.collection.JavaConversions._
import java.util
import scala.io.Source