Add support for flood control