Adding training of KNN models