Update sample graph to include arguments