Extensions to Bayesian Additive Regression Trees