Using K-means for classification