Changed the sampler to NUTS