Setting up the model