Update test_k_mean.jl