Added some tests for the model