Small change to PrepareKMeansInputFile function.