Update data science solution

That's fine, but find a way to represent that without a class-level comment. Even just using `cohort_name`, for instance, would explain better than a comment.