diff --git a/src/main/java/kz/edu/nu/cs/exercise/State1.java b/src/main/java/kz/edu/nu/cs/exercise/State1.java
new file mode 100644
index 0000000..bd159a1
--- /dev/null
+++ b/src/main/java/kz/edu/nu/cs/exercise/State1.java
@@ -0,0 +1,22 @@
+package kz.edu.nu.cs.exercise;
+
+public class State1 extends State{
+	protected StateContext sc;
+    protected boolean accept = false;
+    
+	public State1(StateContext stateContext) {
+		this.sc = stateContext; 
+	}
+	public void actionA() {
+		this.sc.setCurrentState(this.sc.getState2());
+    }
+
+    public void actionB() {
+//    	no change of the state
+    }
+
+    public boolean isAccept() {
+        return this.accept;
+    }
+}
+
diff --git a/src/main/java/kz/edu/nu/cs/exercise/State2.java b/src/main/java/kz/edu/nu/cs/exercise/State2.java
new file mode 100644
index 0000000..7a00fbf
--- /dev/null
+++ b/src/main/java/kz/edu/nu/cs/exercise/State2.java
@@ -0,0 +1,23 @@
+package kz.edu.nu.cs.exercise;
+
+public class State2 extends State{
+	protected StateContext sc;
+    protected boolean accept = false;
+    
+	public State2(StateContext stateContext) {
+		this.sc = stateContext; 
+	}
+	public void actionA() {
+		this.sc.setCurrentState(this.sc.getState3());
+    }
+
+    public void actionB() {
+    	this.sc.setCurrentState(this.sc.getState1());
+    }
+
+    public boolean isAccept() {
+        return this.accept;
+    }
+}
+	
+
diff --git a/src/main/java/kz/edu/nu/cs/exercise/State3.java b/src/main/java/kz/edu/nu/cs/exercise/State3.java
new file mode 100644
index 0000000..c6fda21
--- /dev/null
+++ b/src/main/java/kz/edu/nu/cs/exercise/State3.java
@@ -0,0 +1,22 @@
+package kz.edu.nu.cs.exercise;
+
+public class State3 extends State{
+	protected StateContext sc;
+    protected boolean accept = true;
+    
+	public State3(StateContext stateContext) {
+		this.sc = stateContext; 
+	}
+	public void actionA() {
+//		no change of the state
+    }
+
+    public void actionB() {
+    	this.sc.setCurrentState(this.sc.getState2());
+    }
+
+    public boolean isAccept() {
+        return this.accept;
+    }
+    
+}
diff --git a/src/main/java/kz/edu/nu/cs/exercise/StateContext.java b/src/main/java/kz/edu/nu/cs/exercise/StateContext.java
index 3b154c4..767bc3d 100644
--- a/src/main/java/kz/edu/nu/cs/exercise/StateContext.java
+++ b/src/main/java/kz/edu/nu/cs/exercise/StateContext.java
@@ -14,17 +14,19 @@ public StateContext() {
     public void actionA() {
         // complete this method by
         // delegation to the current state
+    	this.currentState.actionA();
     }
 
     public void actionB() {
         // complete this method
         // delegate to the current state
+    	this.currentState.actionB();
     }
 
     public boolean inAcceptState() {
         // complete this method and return correct value
         // delegate to the current state
-        return false;
+        return currentState.isAccept();
     }
 
     public State getCurrentState() {
@@ -35,4 +37,14 @@ public void setCurrentState(State currentState) {
         this.currentState = currentState;
     }
 
+    public State getState1() {
+    	return state1;
+    }
+    public State getState2() {
+    	return state2;
+    }
+    public State getState3() {
+    	return state3;
+    }
+
 }
diff --git a/src/main/java/kz/edu/nu/cs/exercise/Test.java b/src/main/java/kz/edu/nu/cs/exercise/Test.java
new file mode 100644
index 0000000..efb266f
--- /dev/null
+++ b/src/main/java/kz/edu/nu/cs/exercise/Test.java
@@ -0,0 +1,16 @@
+package kz.edu.nu.cs.exercise;
+
+public class Test {
+
+	public static void main ( String [] args ) {
+		StateContext sc = new StateContext ();
+		sc . actionA ();
+		sc . actionA ();
+//		sc . actionB ();
+		// Is machine in an accept state
+		// after receiving AAB
+		System.out.println(sc.inAcceptState());
+//		sc . inAcceptState ();
+	}
+
+}