Move treatment to treatment controller