modifiy munge and add wraper in model fit