Swimming outfit for 6 month old

Al
Hi! I've signed up for swimming classes for my son but I'm now faced with the swimming trunks and waterproof diapers situation. What set-up is best for pool?