added confirm dialog back
minor clean up
revert changes to original drawTreatments, not used right now