added tests for models