Fixed a few minor problems with the model.