Add test for models.