Changes to the model and controller in order to integrate our code.