Code for cohorts API
Add image and message models