added test coverage for states