Add flow rate control and an init() for extensions
Move init code to init()