Added tests for the model.