Added more unit tests for the models.