K-means implementation with Spark ML