Added unit tests for the models.