Moved runnig proba_test to a separate function.
Added cli interface for plots saving.