Added tests for state classes