Improve logic for cohort visibility labels

TNL-25
That makes sense. I'm happy to rename this so to be more generic that it can evolve to become that function, e.g. ```augment_thread_info```?