5 week old indoor swimming

What would you dress a 5 week old boy for swimming? Just shorts or a full swim costume suggestions please