Updated model version with pre_run