Updated header file