If you replace the height control algorithm with this code you get a little more control over the floating behaviour:
Where the Gain parameter controls the speed of the height control and the Damping controls the softness/damping of the control, a value of 1 means well damped and a value of 0 totally undamped and the mas will oscillate freely.