Updated graph to just have surplus