Parallelized treatment effects code