Worked on the header file