Change flow_graph to top_block