forgot the State class