add flag on units