Made changes to the graph plotting script