My K-Means algorithm implementation on python