Add setOnDismissListener