Adds randomized treatment arm to the final dataset, closes #29