small fixes in mdn trainer