small changes to wordcountbatch