working on the updated model fitting