Fixed custom highlight bug
Updated connection highlights
Removed hardcoded 'in', 'out'