Laid off at 8 months pregnant - legal action

Hi, I was laid off at 8 months pregnant and the company offered me (I am a corporate lawyer) 2 months of severance. I was shocked that I was included in the layoff given my pregnancy, but I had observed some strange behavior towards me in the months leading up to the layoff (after announcing the pregnancy) that made me somewhat suspect something was up. I have the below as examples of circumstantial evidence -

- they downgraded my performance after I announced pregnancy (they claim that they had already made this submission prior to my announcement). my performance review being downgraded was due to "insufficient soft skills." I had been with this company for 3 years prior and always received consistent praise on these alleged "soft skills" as well as 2 promotions.

- the downgraded performance review period was for during a time that my manager herself was on maternity leave (50% of the evaluated period). i was accordingly covering for her for the period she downgraded me for.

- my manager placed me on a PIP leading up to my due date. i shared this information with HR and they had my manager immediately remove the PIP.

I retained an employment lawyer and we wrote a demand letter. The company increased my package to 10 months of severance. Would you stop here or would you sue? My pros for accepting the package are immediate relief/an end to the adversarial process. My cons for accepting the package are the feeling that I did not fully assert my rights/stand up for justice/get adequately financially compensated for the emotional distress of the situation. I am honestly very confused. My lawyer thinks that I have a strong case but also thinks the company would fight hard and drag out the process into a length legal battle (1-3 years plus potential appeals etc). What would you all do?