I would highly recommend swimsuits with flotation devices built in, as it was what my daughter used while learning to swim. It helps to give kids confidence, because they know that they aren't going to sink, so they can concentrate on learning their strokes. The only drawback to these is that some swim lessons will not allow the child to wear them, so it is always best to check with the swim instructor first to see what is allowed.
great job!