How to make it so that when the game starts we can still change the teams in the menu ?

I think this is using the Custom option of the 2nd setting in Island Settings, Teams.
It says using it with Dynamic team size breaks island matchmaking, so maybe set it to split evenly or 1 to 4 players.