Added highlight connected nodes functionality