I think this is the foundations to a really great game. There's definitely some tweaks to be made like being clearer to the player about certain elements of the game (the objective, what happens if you don't end the dive and what is in chests) but you're definitely onto something here!