Add flow control