Add notes to document

1 files changed, 11 insertions, 1 deletions

diff --git a/src/main/java/org/zwobble/mammoth/documents/Note.java b/src/main/java/org/zwobble/mammoth/documents/Note.java
index 1a0d3c0..da6f68f 100644
--- a/src/main/java/org/zwobble/mammoth/documents/Note.java
+++ b/src/main/java/org/zwobble/mammoth/documents/Note.java
@@ -3,11 +3,21 @@ package org.zwobble.mammoth.documents;
 import java.util.List;
 
 public class Note {
+    private final NoteType noteType;
     private final String id;
     private final List<DocumentElement> body;
 
-    public Note(String id, List<DocumentElement> body) {
+    public Note(NoteType noteType, String id, List<DocumentElement> body) {
+        this.noteType = noteType;
         this.id = id;
         this.body = body;
     }
+
+    public NoteType getNoteType() {
+        return noteType;
+    }
+
+    public String getId() {
+        return id;
+    }
 }