Added node selection functionality