prepare_stan_data function

Prepare input data to run the Stan model