Add the State class, and fill the header file