Remove redundent code Set node when adding to graph (do not set owner manually) Tidy up tests after changes.