Simplified graphviz output into one class