Added prototype for State class