Updated header file to match its source file