added token_generator.py import line