Update jags hierarchy to stop theta diverging. Add some plot commands