Update to graph4
Fix bug