Update tokenize.py