Waterfall Chart - Dynamic Height of Total/Subtotals in Deneb

I was hoping someone could provide me with some guidance… I have just discovered Vega and I am trying to build a waterfall chart in Power BI using Deneb that has subtotals and have referenced this example as my starting point which was built using Vega.

I have modified the code so it works with the Power BI dataset and all is great; however, I am looking to dynamically modify the height of just the total/subtotal bars which I haven’t been able to work out. I have tried playing with domainMax for the Y axis but it shrinks all the bars so doesn’t achieve what I am looking for.

Any help or guidance would be greatly appreciated.