Added a few equations to the model