Fixed a bug in K-Means

A bug: keep old centers not correctly assigned