update owt_kmeans.py