Add drawing of connections