Question about trek Marlin 6 suspension upgrade

I have the 2021 model marlin 6 and I feel like I need to upgrade the suspension because the one it came with is crap. How do I look for right suspension? I think it is a semi integrated headset. What should I look for? Thanks.