The states are also displayed in the graph now