This ones a bit different as it's the "water" rather than "air" that is the effector. Slight difference, but big program difference.

Underwater > Air test. When you reach the air you bob on the surface and can move left and right. You can push up+jump to jump out of the water, and down to dive back in.