The witness is a really interesting puzzle game that can be had for not that much.
Or if you are looking for something more actiony then I would recommend remnant: from the ashes or remnant 2. Described as souls like with guns, but they really change up the formula I found with semi random worlds and bosses.
There was a section where you could swim underwater. It's basically on rails though.