Lex, the AI writing tool, is evolving.