Add ability to dismiss dialog