Improve node selection

Clicking on a node will now select it.