Data science - Revision 2

This looks very redundant to the create_cohort method in main.rb, by encapsulating this behavior in the cohort class, you would make your test less brittle, and remove code duplication