Update for graph visualization

- Removed obsolete, complicated and not working code
- Updated to work with latest version of Dagre