Added in the db call for models