Added State class