added the graph for the different treatments