got basic treatment assignment working