Yes, when you’re making a superflat world just hit the customize button.
Then once you do that there should be a text box at the top with some numbers and different block types.
The numbers adjust how many layers of each block, and the name decides which block is used in eavh layer