run_simulation:fit_model function should be separately defined?

[mihaifalcusan]

Open question: should we define fit_model outside and call it for a better overview of the code?