add treatment plans and treatment models and unit test