Add include for new model type