Updating install script to nicely handle nodes