Add the addMockGraph() method