Sandstone and gravel generating in small patches in beaches and deserts

Sandstone should generate in small patches in beaches and deserts(like how nether wart blocks and warped wart blocks do in crimson/warped forests).It would make the terrain look a little more detailed, especially in deserts because it would no longer look a little flat.It would also make sense. As for gravel, it's basically the same thing as for sandstone(I also learned there's a lot of gravel in real life deserts). 

Sandstone and gravel should also generate in badlands(I didn't have the space to write it in the title), but the sandstone should instead be red sandstone(because it matches red sand that generates in badlands)