Better modal dialog controls