Drag and Dropping Grid Based Jigsaw Inventory

So after following the Drag and Drop with UMG tutorial posted by Epic, with everything working properly, they don’t really explain how to do much else with it like item stacking, or different sized items like you’d have in an action RPG like Path of Exile or Diablo. How would one go about doing something like this? I’m thinking that you could just have a variable that tells it how many slots it should take up, and then adjust the button’s size accordingly. Would this work?