Implement update_pi_beta; make variational-batch compile