I liked how the swimming felt, although a natural consequence of the mouse distance determining speed, it was faster to swim horizontally rather than vertically. Maybe you could incorporate that feature into a puzzle-like obstacle course underwater!
