Fixed up the model a little bit.