Add tests for the different sub-models.
Add tests for the different sub-models.