Looked at some examples to make a few changes to the model.