1 files changed, 12 insertions, 6 deletions
diff --git a/src/T2DLL/T2RemoveFavoriteDialog.h b/src/T2DLL/T2RemoveFavoriteDialog.h
index fbcebd3..50e1d00 100644
--- a/src/T2DLL/T2RemoveFavoriteDialog.h
+++ b/src/T2DLL/T2RemoveFavoriteDialog.h
@@ -1,16 +1,22 @@
 #pragma once
 #include "common.h"
+#include "T2Dialog.h"
 
-class T2RemoveFavoriteDialog {
+class AFX_EXT_CLASS T2RemoveFavoriteDialog : public T2Dialog {
 public:
 	T2RemoveFavoriteDialog();
 	virtual ~T2RemoveFavoriteDialog();
+    void InitializeForRemovePeople(T2TowerDoc* inDoc);
+    void InitializeForRemoveTenant(T2TowerDoc* inDoc);
+
 protected:
 	virtual void OnT2Create();
-	virtual void CreateDlgItem(CTokenizer&, T2Dialog::T2DialogDef&);
+	virtual void CreateDlgItem(CTokenizer& inTokenizer, T2DialogDef& inDef);
 	virtual void OnT2Destroy();
-	virtual int OnT2DialogCommand(unsigned int, long);
-public:
-	void InitializeForRemovePeople(T2TowerDoc*);
-	void InitializeForRemoveTenant(T2TowerDoc*);
+	virtual BOOL OnT2DialogCommand(WPARAM inWParam, LPARAM inLParam);
+
+    int mNameType;
+    CWnd *mCloseButton;
+    CWnd *mDeleteButton;
+    T2NameTable *mNameTable;
 };