Add additional tests for the State class