Update MLMTL script for real datasets