add graph visualization stage for testing