small changes for graph