added tests for model