UI - colours correspond to connections

I made the colors correspond to connections. Looks less pretty, but it's
more useful for code analysis.