Add prototype for flow tracking