Adding the model too