Added the header file and fixed the duplicate problem