Updating the cohort.rb model