update token-generator to work