implemented K-means
i think it works?