k_means_recalc now uses argmin instead of loop