1 files changed, 6 insertions, 1 deletions
diff --git a/Project/ControlElement.h b/Project/ControlElement.h
index dedcc80..2a516aa 100644
--- a/Project/ControlElement.h
+++ b/Project/ControlElement.h
@@ -32,11 +32,16 @@ public:
 
     void StartMove(wxPoint2DDouble position);
     void Move(wxPoint2DDouble position);
-    bool Contains(wxPoint2DDouble position) const { return m_rect.Contains(position); }
+    bool Contains(wxPoint2DDouble position) const;
+    
+    bool IsConnected() const { return m_connected; }
+    void SetConnected(bool connected = true) { m_connected = connected; }
 
 protected:
     wxRect2DDouble m_rect;
     NodeType m_nodeType;
+    
+    bool m_connected = false;
 
     wxPoint2DDouble m_moveStartPt;
     wxPoint2DDouble m_movePos;