update TaskGraph.java