Updates to td fit model