Enhances model.dia