Add code to print num nodes