fixed up the header files