Add the flag control