add more input and treatment types