Add tests for Model