New TODO for better total number of nodes calculation.