add first try at a fit_model function