Add some more tests to the model