Very simple Implementation of the k-means algorithm in Python