Adds randomized treatment arm as a graph, closes #28