Added shuffle for output tokens