Added unit test for domain model