Swift version of the original WaterflowView