Data science/mabdeljalil V2

You could use some more tests here.  What if the data is corrupted?  What if there are > 2 cohorts?  What if there is only one?  There are definitely some more cases, including edge cases, that could be tested here.