Notes the addition of treatment arm to graphs