added diet_type to the model