The writing's already pretty good, even Terrence Tao calls it flawless, if you read the slides. That's not where the LLM requires help.