Added unit tests for models.