State class

Fixed the state class return values to return a string