Add equal_total_weight argument for clustering