1 files changed, 11 insertions, 0 deletions
diff --git a/Swiften/Elements/WhiteboardPayload.h b/Swiften/Elements/WhiteboardPayload.h
index 3ad1706..c61e4a5 100644
--- a/Swiften/Elements/WhiteboardPayload.h
+++ b/Swiften/Elements/WhiteboardPayload.h
@@ -9,12 +9,14 @@
 #include <string>
 
 #include <Swiften/Elements/Payload.h>
+#include <Swiften/Whiteboard/Elements/WhiteboardElement.h>
 
 namespace Swift {
 	class WhiteboardPayload : public Payload {
 	public:
 		typedef boost::shared_ptr<WhiteboardPayload> ref;
 
+	public:
 		enum Type {Data, SessionRequest, SessionAccept, SessionTerminate};
 
 		WhiteboardPayload(Type type = WhiteboardPayload::Data) : type_(type) {
@@ -36,8 +38,17 @@ namespace Swift {
 			type_ = type;
 		}
 
+		WhiteboardElement::ref getElement() const {
+			return element_;
+		}
+
+		void setElement(WhiteboardElement::ref element) {
+			element_ = element;
+		}
+
 	private:
 		std::string data_;
 		Type type_;
+		WhiteboardElement::ref element_;
 	};
 }