Add working model fit