Nice concept, and well executed! Using those three states allows for some pretty interesting challenges, I'm looking forwards to an extended version with more levels and some twists :)
I expected the steam stage to make you rise automatically, it could add an interesting constraint - though I'd still allow to fight off this pull so that you can still go down (but slowlier) while in steam form.