added the class declaration to the header files