Added test classes for two models.