Cleanup `DomainAwareDataset` API

[kachayev]

• review methods and naming convention, make sure all methods are necessary
• write docstrings
• test cases, specifically to cover the functionality of adding new domains, getting domain and selecting multiple domains
• test cases to cover failure modes, like overlapping names