Changing array to use a hash and adding cohort information