Updated the header file