Add new Spout graph for treatment arm, fix #284