Add image option to basic node-link diagram

Alternative to having a separate visualization, cleaner and simpler