remove unused imports
remove unneeded logging info
third command line argument to specify number of nodes