Added tests for additional states.