Adding optional flag