K-means algorithm implemented in Go (golang)