add flag function