Changed stateModel, and updated the TestModel to work with what I did.