Add flag function