Added further DMLT support, fixed bug in second level model