Update CoarseGrainedTree.java

Add comments and some adjustments