re-add kmeans classification code