Moved visualization to separate module
Switched to pygraphviz