Made the dialog return callback more flexible