fix(core): ensure element is available during draw

By adding a shape / connection to the registry before updating it we
make sure it is available during update / layout.
chore(layout): cleanup unused code