added graph visualization