top right, update legend to show cohort: <img width="1123" height="344" alt="Image" src="https://github.com/user-attachments/assets/dbc452df-c4c1-4c48-8e6f-ea17c2954448" />
top right, update legend to show cohort: