Mini Menu in Creative Inventory

The Creative Inventory isn't great in Java or Bedrock. I have one idea to improve it, it is when you middle-click, option button on controller, or long-press on PE, a square below and above appear. (Legacy console when crafting has the square above and below if you want to know what it would look like) These items/blocks would be related to each other. E.g. middle-click oak planks, see other oak blocks (not other types of planks). 

If you wanted more options you could add squares to sides of UI to change shape of block, so down is different planks.

Also all these option could be togglable in settings.