Spark Streaming implementation of Online KMeans