Updated the model specs