Treasury threatens to withhold funds for more counties