ADd more dm unit tests