treatment assignment mechanism