Add variable number_of_non_leaf_nodes and comment to increase readability.