Legion 2 ssd upgrade...

My legion go 2 is arrving on monday, im looking to upgrade to either 2tb or 4tb.

What would you recommend?