--- a/DboxMain.h
+++ b/DboxMain.h
@@ -200,6 +200,8 @@
                         const bool bIsEdit, uuid_array_t &base_uuid, int &ibasedata);
   void GetBaseUUID(const uuid_array_t &alias_uuid, uuid_array_t &base_uuid)
   {m_core.GetBaseUUID(alias_uuid, base_uuid);}
+  int GetEntryImage(const CItemData &ci);
+  HICON GetEntryIcon(const int nImage) const;
 
   //{{AFX_DATA(DboxMain)
   enum { IDD = IDD_PASSWORDSAFE_DIALOG };
@@ -327,7 +329,6 @@
   void SetDCAText();
   void SortListView();
   void UpdateBrowseURLSendEmailButton(const bool bIsEmail);
-  int GetEntryImage(CItemData ci);
   void SetEntryImage(const int &index, const int nImage, const bool bOneEntry = false);
   void SetEntryImage(HTREEITEM &ti, const int nImage, const bool bOneEntry = false);
 
@@ -464,6 +465,9 @@
   afx_msg void OnToolBarClearFind();
   afx_msg void OnHideFindToolBar();
 
+  afx_msg void OnDrawItem(int nIDCtl, LPDRAWITEMSTRUCT lpDrawItemStruct);
+  afx_msg void OnMeasureItem(int nIDCtl, LPMEASUREITEMSTRUCT lpMeasureItemStruct);
+
 #if _MFC_VER > 1200
   afx_msg BOOL OnOpenMRU(UINT nID);
 #else