Add tests for models.