K-means Algorithm written in Python