Improved token pool memory allocation