Added code to highlight node connections.
Node reference names are now automatically found from the class names.