FIX handling of treatment events
ADDS more treatment options to shiny UI