Different levels per branch

Hi guys,
I wanted to ask what is the way to pile up different amount of volumes depending on the branch. What I need is to have 3 levels of the red ones, 6 levels of the violet ones, and 9 of the green ones without changing the tree structure (like in the screenshot), and I can’t figure out what is wrong with the definition I am attaching here. I would appreciate any help, thank you!!

From having this ground floor (only ground floor level),


I now want to have this amount of floors/levels,

pile up volumes.gh (381.1 KB)

forum_pile.gh (383.4 KB)

from what I can understand, this seems to be what you’re after