add rstan code