Update the header files too