save 100% There's 3 lines of token_dir. Do we replace the entire thing with "if True" or do we just add it in the end of each line?
