setup_S_model function

Setup survival part for outcome m