Change the UI depending on treatment.