diff --git a/src/main/java/MyStack.java b/src/main/java/MyStack.java index d88e45043f6e81f0a3a36ceca83962ed1692e9e4..3e2386f5e1b96c98badcd1078110cace9cfa90a9 100644 --- a/src/main/java/MyStack.java +++ b/src/main/java/MyStack.java @@ -1,3 +1,5 @@ +import java.util.EmptyStackException; + public class MyStack { private Node top; @@ -16,6 +18,7 @@ public class MyStack { } public Object pop() { + if (size() == 0) throw new EmptyStackException(); Object payload = top.getPayload(); top = top.getPrevious(); return payload;