What is the best swimming attire for my 6-month old son? This will be his first time to swim :)

I had the normal one, but if you scared baby will get cold can buy those with warmer one. Can be found in mothercare or swimming costume shop.