New UX for modeling: Formula broken down into lines & color coded

In the new modeling UX, it would be great if the formula is broken down into lines and indented when different parts of the formulas are in use (especially for multiple IF statements). Right now, even though the colors change per formula, it's hard to find where the breaks are.

Additionally, right now every IF statement shows with the same color, while it would be great if each of them could have their own color, so we can easily see which THEN and ELSE refer to that IF.

For reference, this would be similar to how the A+ add in shows the formulas. Screenshot attached for reference