1 files changed, 18 insertions, 7 deletions
diff --git a/src/256CheckDialog.h b/src/256CheckDialog.h
index f95925b..5ad0f81 100644
--- a/src/256CheckDialog.h
+++ b/src/256CheckDialog.h
@@ -1,18 +1,29 @@
 #pragma once
 #include "common.h"
+#include "resource.h"
 
-class C256ColorDialog : public CDialog {
-    DECLARE_MESSAGE_MAP()
-
+class C256CheckDialog : public CDialog {
 public:
-    C256ColorDialog(CWnd *pParentWnd);
-    virtual int DoModal();
+    C256CheckDialog(CWnd *pParent = NULL);
 
-protected:
+    //{{AFX_DATA(C256CheckDialog)
+    enum { IDD = IDD_256CHECK };
+    CButton mButton;
+    //}}AFX_DATA
+
+    //{{AFX_VIRTUAL(C256CheckDialog)
+    public:
+    virtual int DoModal();
+    protected:
     virtual void DoDataExchange(CDataExchange *pDX);
     virtual void OnOK();
     virtual void OnCancel();
+    //}}AFX_VIRTUAL
+
+protected:
     void Save();
 
-    CButton mButton;
+    //{{AFX_MSG(C256CheckDialog)
+    //}}AFX_MSG
+    DECLARE_MESSAGE_MAP()
 };