Updated header file to expand the container