There’s multiple paths in the game that open up slightly different content. There’s a critical point when you can choose to take Sophia with you, or to go it alone with either thinking (harder puzzles) or fighting.

The balloon is shortly before the submarine on that path (I believe it’s the sophia path)