Fix k-means and the example of bad init means
Finish the third problem of hw5