Replaced main with use_jags