Added test class for State