Kmeans training improved

Kmeans now takes features from more than one class (to be checked)