Updated Token_Generator function