Yes, that would be great. And for those who become trad wives later in life to raise children even for several years, the prenup should include a stipulation of being compensated for taking a break from work, missing opportunities for several raises and slowing down the income increase over time. That's unless the spouse is willing to share the parental leave and lessen the financial hit; some good fathers who want to spend time with their babies do so willingly!
I mean, the trad wife scenario may work out for some. Sadly, considering the divorce rates and stories of former trad wives, often it does not.