The looked so cute in the water