Adding multi-allocation logic