add some code to the header file