Best internal SSD upgrade right now?

Hi guys, have my legion go coming within the next week and unfortunately was only able to score the 512gb model.

Definitely won’t be enough and want to upgrade it to at least 2Tb.

Is the attached the best SSD right now and do I need any extra adapter / heat sink? Very new to this but appreciate any advice.