Updated the model initialization process.