Updated the graph creation script

The functionality should be moved to statistics though