Add FlagSet tests