A few implementation fixes of the NUTS algorithm
Scaled the regularization down to prevent biased estimates when the (co)variance is small