Add get_flows function.