1 files changed, 11 insertions, 4 deletions
diff --git a/src/T2PoolView.h b/src/T2PoolView.h
index 7b24545..9f341bc 100644
--- a/src/T2PoolView.h
+++ b/src/T2PoolView.h
@@ -1,20 +1,27 @@
 #pragma once
 #include "common.h"
+#include "resource.h"
 
 class T2PoolView : public CFrameWnd {
     DECLARE_DYNCREATE(T2PoolView)
-    DECLARE_MESSAGE_MAP()
 
 public:
     T2PoolView();
     virtual ~T2PoolView();
     void SetPool(T2Pool *inPool);
 
+    T2Pool *mPool;
+
+    //{{AFX_VIRTUAL(T2PoolView)
+    //}}AFX_VIRTUAL
+
 protected:
+    void Draw(CDC *pDC, CPoint *inPt, T2PeopleTimeZoneList **outTimeZoneList, T2PeopleTypeArray **outPeopleTypeArray, int *outUnknown);
+
+    //{{AFX_MSG(T2PoolView)
     afx_msg BOOL OnEraseBkgnd(CDC *pDC);
     afx_msg void OnVScroll(UINT nSBCode, UINT nPos, CScrollBar *pScrollBar);
     afx_msg void OnLButtonDown(UINT nFlags, CPoint point);
-    void Draw(CDC *pDC, CPoint *inPt, T2PeopleTimeZoneList **outTimeZoneList, T2PeopleTypeArray **outPeopleTypeArray, int *outUnknown);
-
-    T2Pool *mPool;
+    //}}AFX_MSG
+    DECLARE_MESSAGE_MAP()
 };