Update klp_kmeans.py

batch indexing fix