How do i create a swamp biome in the newest thing

Im trying to add a flooded rainforest biome to the game but i need to know how to make it so the water is only on flat surfaces to prevent flooding. Thank you for helping!