add dml test code