diff --git a/digesttreescene.h b/digesttreescene.h
index 29d9177..9d9dee4 100644
--- a/digesttreescene.h
+++ b/digesttreescene.h
@@ -36,6 +36,7 @@
   bool removeNode(const QString prefix);
   void plot(QString digest);
   QStringList getRosterList();
+  void setCurrentPrefix(QString prefix) {m_currentPrefix = prefix;}
 
 signals:
   void replot();
@@ -52,6 +53,7 @@
   Roster m_roster;
   QGraphicsTextItem *m_rootDigest; 
   DisplayUserPtr previouslyUpdatedUser;
+  QString m_currentPrefix;
 
 };