Add tests for domain model