Improve K-means wizard