Added more plots to plot_state_statistics.py.