Survey builder: does this flow make sense?

I'm adding a new feature to a survey builder product. It allows the user to choose the appropriate tone of voice for a phrase; the user would be able to decide which phrasing would best fit their survey.

Below is a flow for how this could work. Any insights on what to improve /consider?

Four screenshots that show: 1. User sees a phrase; 2. User clicks on the phrase to see options based on tone of voice; 3. User selects an option; user sees the original phrase replaced by the option