Added states enum and JUnit test case