That should probably work yeah, half of what I’m keeping in mind is generalising to an arbitrary number of U_i but I can just lump that into a for loop and work through like you’ve presented. Just me overcomplicating a simple problem lol 
Many thanks!