Improve dismiss function