Added user input for node generator functionality