Add UI parameter to handler function