Swim safety

What are we using when taking our 15 month olds swimming? Swim float/swim vest/arm bands? We’re more taking her swimming to splash about and get comfortable in the water currently over actually swimming although we are planning to start some lessons .. but trying to find the best thing to put on her to keep her safe whilst in the water with us.