some slight changes to the graph controller