Fixed the polya gamma sampler