Pupils have been learning all about keeping safe in water...