diff --git a/patterns/design-patterns-2/pom.xml b/patterns/design-patterns-2/pom.xml
index 392035ab84..574edecb36 100644
--- a/patterns/design-patterns-2/pom.xml
+++ b/patterns/design-patterns-2/pom.xml
@@ -31,6 +31,12 @@
             <artifactId>commons-lang3</artifactId>
             <version>${commons-lang3.version}</version>
         </dependency>
+        <dependency>
+            <groupId>org.assertj</groupId>
+            <artifactId>assertj-core</artifactId>
+            <version>${assertj.version}</version>
+            <scope>test</scope>
+        </dependency>
     </dependencies>
     
     <properties>
@@ -38,5 +44,6 @@
         <maven.compiler.source>1.8</maven.compiler.source>
         <maven.compiler.target>1.8</maven.compiler.target>
         <intellij.annotations.version>16.0.2</intellij.annotations.version>
+        <assertj.version>3.12.2</assertj.version>
     </properties>
 </project>
diff --git a/patterns/design-patterns-2/src/main/java/com/baeldung/memento/TextEditor.java b/patterns/design-patterns-2/src/main/java/com/baeldung/memento/TextEditor.java
index cf08e2b226..37ac962773 100644
--- a/patterns/design-patterns-2/src/main/java/com/baeldung/memento/TextEditor.java
+++ b/patterns/design-patterns-2/src/main/java/com/baeldung/memento/TextEditor.java
@@ -9,6 +9,14 @@ public class TextEditor {
         this.textWindow = textWindow;
     }
 
+    public void write(String text) {
+        textWindow.addText(text);
+    }
+
+    public String print() {
+        return textWindow.getCurrentText();
+    }
+
     public void hitSave() {
         savedTextWindow = textWindow.save();
     }
diff --git a/patterns/design-patterns-2/src/main/java/com/baeldung/memento/TextWindow.java b/patterns/design-patterns-2/src/main/java/com/baeldung/memento/TextWindow.java
index 5fdfda1998..08778561b0 100644
--- a/patterns/design-patterns-2/src/main/java/com/baeldung/memento/TextWindow.java
+++ b/patterns/design-patterns-2/src/main/java/com/baeldung/memento/TextWindow.java
@@ -3,11 +3,13 @@ package com.baeldung.memento;
 public class TextWindow {
 
     private StringBuilder currentText;
-    private Coordinates cursorPosition;
 
     public TextWindow() {
         this.currentText = new StringBuilder();
-        this.cursorPosition = new Coordinates(0, 0);
+    }
+
+    public String getCurrentText() {
+        return currentText.toString();
     }
 
     public void addText(String text) {
@@ -20,11 +22,5 @@ public class TextWindow {
 
     public void restore(TextWindowState save) {
         currentText = new StringBuilder(save.getText());
-        cursorPosition = atTextEnd();
-    }
-
-    private Coordinates atTextEnd() {
-        String[] lines = currentText.toString().split("\n");
-        return new Coordinates(lines[lines.length - 1].length(), lines.length);
     }
 }