improve flow graph a bit