trtAssign function

Assign treatment