Add initial version of fit_sota_model