Add return graph