K-Means algorithm implementation in Go