Lazy Connect with Menu (Shift+Alt+RMB) improvement

i am a big fan of node wrangler and Lazy Connect feature is very handy to connect nodes that are far out of each other. However, when there is multiple sockets on both sides - the Menus are not so intuitive to choose from. My suggestion for improvement:

1. add colorful sockets - right to the text for outputs and left to the text for inputs (just like on the nodes itself)
2. make both output and input on single menu - so you can choose them in no particular order.

see picture below: