Automation Categories - Add Expand / Collapse

Having Categories available on the automation dialogue is very nice, could the expand / collapse feature be added to the category titles? Something similar to the icon used in automations to expand/collapse details:

image