Spark app that divides a RDD into train/test set