Add flag to unit