Speed up K-Means clustering by porting Python implementation to Mojo🔥