allot_trts function

Define allotment of treatments to units