Would it be beneficial if we also took which number of cluster to update? That way we can remove the check that we only have 1 cluster.