Update token_code.rb