Added functionality to show connection between nodes