added a test for the model