Fetching SelfEndorseCertificate works
diff --git a/src/warningdialog.h b/src/warningdialog.h
index 7b3125a..a2f3481 100644
--- a/src/warningdialog.h
+++ b/src/warningdialog.h
@@ -1,3 +1,13 @@
+/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil -*- */
+/*
+ * Copyright (c) 2013, Regents of the University of California
+ *                     Yingdi Yu
+ *
+ * BSD license, See the LICENSE file for more information
+ *
+ * Author: Yingdi Yu <yingdi@cs.ucla.edu>
+ */
+
 #ifndef WARNINGDIALOG_H
 #define WARNINGDIALOG_H
 
@@ -9,14 +19,22 @@
 
 class WarningDialog : public QDialog
 {
-    Q_OBJECT
+  Q_OBJECT
 
 public:
-    explicit WarningDialog(QWidget *parent = 0);
-    ~WarningDialog();
+  explicit WarningDialog(QWidget *parent = 0);
+
+  ~WarningDialog();
+                  
+  void
+  setMsg(const std::string& msg);
+
+private slots:
+  void
+  onOkClicked();
 
 private:
-    Ui::WarningDialog *ui;
+  Ui::WarningDialog *ui;
 };
 
 #endif // WARNINGDIALOG_H