Quick changes to printState method