added train_dpm apparently it wasn't tracked before