add return to dialog