Add class for neato layout graphviz plot and fix a bug
removed unused code