Writing the K-Means algorithm in Rust