thought I might be able to do it quickly but it seems one has to understand how to reimplement it with the GraphStage.