New code to perform inference on treatment effects