When we put both my young kids through swim lessons, they were taught with "bubbles", little foam blocks that are strapped to the middle of the back, and the number of bubbles are reduced as their proficiency increases. That left their arms and chest free to work on actual swimming form. When working on kicks, they held on to little floating dumbell bars.