* Fix graph clenup * Refactoring graph clean up function * Change wa comment Co-authored-by: Your Name <you@example.com>