Add tests for model.