Adding initial draft of k-means