Update main.py

Implemented GraphViz module to construct DOT code for the GraphViz graph generator.