Add the necessary parameters to the Model classes