Category icons on shortcut pop-up for transactions

One of my favourite features, and the reason I discovered Budget Flow, is the ability to automate expense adding with esch apple pay purchase.

Currently the pop-up that asks you to select the category of that transaction does not include any icon so it’s a bit dificult to find the one you want each time because its not visual:

However, if there was a way to incorporate the category icons (the ones below) at the start of each row, that’d be awesome and a super nice upgrade :)

Thanks