split model fitting to separate section
update the future work