diff --git a/src/warningdialog.cpp b/src/warningdialog.cpp
index 135403c..6c7eb8a 100644
--- a/src/warningdialog.cpp
+++ b/src/warningdialog.cpp
@@ -1,11 +1,26 @@
+/* -*- 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>
+ */
+
 #include "warningdialog.h"
 #include "ui_warningdialog.h"
 
+using namespace std;
+
 WarningDialog::WarningDialog(QWidget *parent) :
     QDialog(parent),
     ui(new Ui::WarningDialog)
 {
-    ui->setupUi(this);
+  ui->setupUi(this);
+
+  connect(ui->okButton, SIGNAL(clicked()),
+	  this, SLOT(onOkClicked()));
 }
 
 WarningDialog::~WarningDialog()
@@ -13,6 +28,14 @@
     delete ui;
 }
 
+void
+WarningDialog::setMsg(const string& msg)
+{ ui->message->setText(QApplication::translate("WarningDialog",msg.c_str(), 0, QApplication::UnicodeUTF8)); }
+
+void
+WarningDialog::onOkClicked()
+{ this->hide(); }
+
 #if WAF
 #include "warningdialog.moc"
 #include "warningdialog.cpp.moc"